from aiogram.types import KeyboardButton
from aiogram.utils.keyboard import ReplyKeyboardBuilder


def menu():
    builder = ReplyKeyboardBuilder()
    for text in [
        "Настройки",
        "Запланированные посты",
        "Запустить рассылку",
    ]:
        builder.add(KeyboardButton(text=text))
    builder.adjust(2)

    return builder.as_markup(resize_keyboard=True)


def admin_menu():
    builder = ReplyKeyboardBuilder()
    for text in [
        "Настройки",
        "Запланированные посты",
        "Запустить рассылку",
        "Рассылка по пользователям",
    ]:
        builder.add(KeyboardButton(text=text))
    builder.adjust(2)

    return builder.as_markup(resize_keyboard=True)


def add_more_kb():
    builder = ReplyKeyboardBuilder()
    for text in [
        "Сделать еще один пост",
        "Завершить работу",
    ]:
        builder.add(KeyboardButton(text=text))
    builder.adjust(2)

    return builder.as_markup(resize_keyboard=True)
